model_id_or_path=$1

if [ ! -z "$model_id_or_path" ];then
    swift sft \
        --model Qwen/Qwen2.5-7B-Instruct \
        --model_id_or_path output/qwen2_5-0_5b-instruct/${model_id_or_path} \
        --dataset data.json \
        --num_train_epochs 5 \
        --train_type lora \
        --output_dir output \
        --eval_steps 200
fi

if [ -z "$model_id_or_path" ];then
    swift sft \
        --model Qwen/Qwen2.5-7B-Instruct \
        --dataset data.json \
        --num_train_epochs 10 \
        --train_type lora \
        --output_dir output \
        --eval_steps 200
fi
